package com.cg.leetcode;

import org.junit.Test;

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Deque;
import java.util.List;

/**
 * 589.N 叉树的前序遍历
 *
 * @program: LeetCode->LeetCode_589
 * @author: cg
 * @create: 2022-04-21 14:32
 **/
public class LeetCode_589 {

    @Test
    public void test589() {
        Node root = new Node(1);
        Node node2 = new Node(2);
        Node node3 = new Node(3);
        Node node4 = new Node(4);
        Node node5 = new Node(5);
        Node node6 = new Node(6);
        root.neighbors = new ArrayList<>();
        root.neighbors.add(node3);
        root.neighbors.add(node2);
        root.neighbors.add(node4);
        node3.neighbors = new ArrayList<>();
        node3.neighbors.add(node5);
        node3.neighbors.add(node6);
        System.out.println(preorder(root));
    }

    /**
     * 给定一个 n 叉树的根节点  root ，返回 其节点值的 前序遍历 。
     * <p>
     * n 叉树 在输入中按层序遍历进行序列化表示，每组子节点由空值 null 分隔（请参见示例）。
     * <p>
     * 示例 1：
     * 输入：root = [1,null,3,2,4,null,5,6]
     * 输出：[1,3,5,6,2,4]
     * <p>
     * 示例 2：
     * 输入：root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
     * 输出：[1,2,3,6,7,11,14,4,8,12,5,9,13,10]
     * <p>
     * 提示：
     * 节点总数在范围 [0, 10^4]内
     * 0 <= Node.val <= 10^4
     * n 叉树的高度小于或等于 1000
     * <p>
     * 进阶：递归法很简单，你可以使用迭代法完成此题吗?
     *
     * @param root
     * @return
     */
    public List<Integer> preorder(Node root) {
        // 迭代
        List<Integer> res = new ArrayList<>();
        if (root == null) {
            return res;
        }
        Deque<Node> stack = new ArrayDeque<>();
        stack.push(root);
        while (!stack.isEmpty()) {
            Node node = stack.pop();
            res.add(node.val);
            for (int i = node.neighbors.size() - 1; i >= 0; i--) {
                stack.push(node.neighbors.get(i));
            }
        }
        return res;
    }

    /*public List<Integer> preorder(Node root) {
        // 递归
        List<Integer> res = new ArrayList<>();
        if (root != null) {
            dfs(root, res);
        }
        return res;
    }

    private void dfs(Node root, List<Integer> list) {
        list.add(root.val);
        for (Node neighbor : root.neighbors) {
            dfs(neighbor, list);
        }
    }*/

}
